from langchain_core.output_parsers import JsonOutputParser
#from langchain_deepseek import ChatDeepSeek
#from zhipuai import ZhipuAI
from langchain_community.chat_models import ChatZhipuAI
from langchain.prompts import PromptTemplate
from dotenv import load_dotenv
import os

load_dotenv()

# 定义输出格式
parser = JsonOutputParser()
prompt = PromptTemplate.from_template("请返回JSON格式的用户信息：{query}")
# 调用大模型并解析
#llm = ChatDeepSeek(model="deepseek-chat")
#llm = ZhipuAI(api_key=os.getenv("ZHIPUAI_API_KEY"))
llm = ChatZhipuAI(api_key=os.getenv("ZHIPUAI_API_KEY"))
output = llm.invoke(prompt.format(query="用户ID 123"))
# 从 AIMessage 中提取内容
parsed_output = parser.parse(output.content)
print(parsed_output)
